| Name | Version | Summary | date |
| jaxtyping |
0.2.38 |
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. |
2025-02-13 22:02:20 |
| cryojax |
0.3.3 |
Cryo-EM image simulation and analysis powered by JAX |
2025-01-15 14:57:43 |
| optimistix |
0.0.10 |
Nonlinear optimisation in JAX and Equinox. |
2025-01-03 23:48:28 |
| diffrax |
0.6.2 |
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX. |
2024-12-24 16:50:08 |
| equinox |
0.11.11 |
Elegant easy-to-use neural networks in JAX. |
2024-12-24 12:07:55 |
| lineax |
0.0.7 |
Linear solvers in JAX and Equinox. |
2024-10-21 10:54:53 |
| flowjax |
16.0.0 |
Easy to use distributions, bijections and normalizing flows in JAX. |
2024-10-17 11:00:06 |
| seasons-cal |
1.0.2 |
None |
2024-10-15 15:21:15 |
| distreqx |
0.0.1 |
Lightweight JAX library of probability distributions and bijectors. |
2024-09-15 06:13:45 |